import socket


def main():
    # 创建套接字
    tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 获取服务器 IP port
    dest_ip = input("请输入要下的服务器IP：")
    dest_port = int(input("请输入要下载的服务器port:"))

    # 连接服务器载
    tcp_socket.connect((dest_ip, dest_port))
    # 获取下载的文件名字
    download_file_name = input("请输入要下载的文件名：")
    # 将文件名字发送到服务器
    tcp_socket.send(download_file_name.encode("utf-8"))
    # 接收文件中的数据
    recv_data = tcp_socket.recv(1024)  # 1024 == 1k
    if recv_data:
        # 保存接收到的数据到另一个文件中
        with open("[新]" + download_file_name, "wb") as f:
            f.write(recv_data)
    # 关闭套接字
    else:
        print("没有要下载的文件(%s)" % download_file_name)
    tcp_socket.close()


if __name__ == "__main__":
    while True:
        print("下载功能请输入：1")
        print("退出功能请输入：2")
        x = input("请输入：")
        if x == "1":
            main()
        if x == "2":
            break
